/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.lvyh.lightframe.sharding.client.parser;

import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser;
import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlOutputVisitor;
import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor;

import java.util.List;

/**
 * Created by lvyun on 2021/10/31.
 */
public class SimpleSqlParser {

    public static String getConvertedSql(String tableName, String newTableName, String sql) {
        MySqlOutputVisitor output = new MySqlOutputVisitor(new StringBuilder());
        MySqlStatementParser parser = new MySqlStatementParser(sql);

        MySqlSchemaStatVisitor schema = new MySqlSchemaStatVisitor();
        List<SQLStatement> sqlStatements = parser.parseStatementList();
        for (SQLStatement sqlStatement : sqlStatements) {
            sqlStatement.accept(schema);
            if (schema.containsTable(tableName)) {
                TableNameModifyASTVisitor tableNameModifyASTVisitor = new TableNameModifyASTVisitor(tableName, newTableName);
                sqlStatement.accept(tableNameModifyASTVisitor);
            }
            sqlStatement.accept(output);
        }

        return output.getAppender().toString().replaceAll("\n", " ").replaceAll("\t", " ");
    }

    public static String getConvertedSql(String tableName, int index, String sql) {
        MySqlOutputVisitor output = new MySqlOutputVisitor(new StringBuilder());
        MySqlStatementParser parser = new MySqlStatementParser(sql);

        MySqlSchemaStatVisitor schema = new MySqlSchemaStatVisitor();
        List<SQLStatement> sqlStatements = parser.parseStatementList();
        for (SQLStatement sqlStatement : sqlStatements) {
            sqlStatement.accept(schema);
            if (schema.containsTable(tableName)) {
                TableNameModifyASTVisitor tableNameModifyASTVisitor = new TableNameModifyASTVisitor(tableName, tableName + index);
                sqlStatement.accept(tableNameModifyASTVisitor);
            }
            sqlStatement.accept(output);
        }

        return output.getAppender().toString().replaceAll("\n", " ").replaceAll("\t", " ");
    }

}
